package top.milkbox.config;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * MyBatis的高级配置信息类
 */
public class MyBatisPlusConfiguration {

    /**
     * 配置内部分页拦截器
     * 用来指定方言，当然也可以省略
     *
     * @return PaginationInnerInterceptor
     */
    @Bean
    public PaginationInnerInterceptor paginationInnerInterceptor() {
        // 对于单一数据库类型来说,都建议配置该值,避免每次分页都去抓取数据库类型
        return new PaginationInnerInterceptor(DbType.MYSQL);
    }

    /**
     * 配置 MybatisPlus 的插件
     *
     * @param paginationInnerInterceptor 内部分页拦截器
     * @return MybatisPlusInterceptor
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor(
            PaginationInnerInterceptor paginationInnerInterceptor
    ) {
        MybatisPlusInterceptor mybatisPlusInterceptor = new MybatisPlusInterceptor();
        // 将内部分页插件添加到 MybatisPlus 插件中
        mybatisPlusInterceptor.addInnerInterceptor(paginationInnerInterceptor);
        // 将“攻击 SQL 阻断解析器”天加到 MyBatisPlus 插件中，防止全表更新与删除
        // 仅用于测试环境
        // mybatisPlusInterceptor.addInnerInterceptor(new BlockAttackInnerInterceptor());
        // 配置乐观锁插件
        // mybatisPlusInterceptor.addInnerInterceptor(new OptimisticLockerInnerInterceptor());
        return mybatisPlusInterceptor;
    }
}
